package com.example.riceexambackend.common.utils;

import com.example.riceexambackend.common.errorcode.CommonErrorCodeConstants;
import com.example.riceexambackend.common.exception.GlobalException;
import io.jsonwebtoken.Claims;
import io.jsonwebtoken.JwtParserBuilder;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.io.Decoders;
import io.jsonwebtoken.security.Keys;

import javax.crypto.SecretKey;
import java.util.Date;
import java.util.Map;

/**
 * JWT工具类
 * @author ricejson
 */
public class JWTUtil {

    private static final Long EXPIRATION_TIME = (30L * 60 * 1000); //
    private static final String secretString = "dVnsmy+SIX6pNptQdeclDSJ26EMSPEIhvZYKBTTug4k="; // 密钥
    private static final SecretKey KEY = Keys.hmacShaKeyFor(Decoders.BASE64.decode(secretString));

    /**
     * 生成令牌
     * @param claim 载荷参数
     * @return 令牌
     */
    public static String genToken(Map<String, Object> claim) {
        return Jwts.builder()
                .addClaims(claim) // 自定义载荷
                .setIssuedAt(new Date()) // 设置签发时间
                .setExpiration(new Date(System.currentTimeMillis() + EXPIRATION_TIME))
                .signWith(KEY) // 签名算法
                .compact();

    }

    /**
     * 解析token
     * @param token 令牌
     */
    public static Claims parseJWT(String token) {
        Claims body = null;
        try {
            // 创建解析器
            JwtParserBuilder parserBuilder = Jwts.parserBuilder().setSigningKey(KEY);
            // 解析token
            body = parserBuilder.build().parseClaimsJws(token).getBody();
        } catch (Exception e) {
            throw new GlobalException(CommonErrorCodeConstants.UNKNOWN_ERROR);
        }
        return body;
    }
}
